import asyncio
from tortoise import Tortoise
from tool.sessing import TORTOISE_ORM
from models.m import NovelCategory, Novel, Chapter, AuditStatus


async def main():
    await Tortoise.init(config=TORTOISE_ORM)
    await Tortoise.generate_schemas()

    # 启用所有分类
    await NovelCategory.all().update(is_active=True)

    # 小说、章节审核通过
    await Novel.filter().update(audit_status=AuditStatus.APPROVED)
    await Chapter.filter().update(audit_status=AuditStatus.APPROVED)

    # 可选：设置部分推荐，方便前端显示
    await Novel.filter(is_featured=False).limit(5).update(is_featured=True)

    print("[OK] 已将分类启用、小说/章节设为已通过，并设置部分推荐。")
    await Tortoise.close_connections()


if __name__ == "__main__":
    asyncio.run(main())


